/*
 * Copyright 北京航空航天大学  @ 2015 版权所有
 */
package com.buaa.edu.leetcode.algorithm.twopoint;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * <p>
 * 三数之和为0
 *  For example, given array S = {-1 0 1 2 -1 -4},
 *  A solution set is: (-1, 0, 1) (-1, -1, 2)
 *  结果需要排序
 * </p>
 * 
 * @author towan
 * @email tongwenzide@163.com
 * @time 2015年5月31日
 */
public class ThreeSum {
    //时间复杂度为n^2主要花在移动首尾指针上
    public List<List<Integer>> threeSum(int []nums) {
        List<List<Integer>> results = new ArrayList<>();
        if(nums==null || nums.length<3){
            return results;
        }
        int len = nums.length;
        Arrays.sort(nums);
        
        for(int i=0;i<len-2;i++){
            if(nums[i]>0){
                break;//大于0没必要
            }
            if(i>0 && nums[i]==nums[i-1]){
                continue;//重复没必要
            }
            int start = i+1;
            int end = len-1;
            while(start<end){
                int sum = nums[i]+nums[start]+nums[end];
                if(sum<0){
                    start++;
                }else if(sum>0){
                   end--; 
                }else{//找到记录
                    List<Integer> res = new ArrayList<>(3);
                    res.add(nums[i]);
                    res.add(nums[start]);
                    res.add(nums[end]);
                    results.add(res);
                    //忽略可能相同的结果
                    do{
                        start++;
                    }while(start<end && nums[start]==nums[start-1]);
                    do{
                        end--;
                    }while(start<end && nums[end]==nums[end+1]);
                }
            }
        }
        
        return results;
    }
    public static void main(String[] args) {
        int []nums = {-1,0,1,2,-1,-4};
        System.err.println(new ThreeSum().threeSum(nums));
    }
}
